import imgaug.augmenters as iaa
from .img_aug import ImgAug


class PadSquare(ImgAug):
    def __init__(
        self,
    ):
        self.augmentations = iaa.Sequential(
            [iaa.PadToAspectRatio(1.0, position="center-center").to_deterministic()]
        )
